package com.wmx.mp.handler;

import cn.hutool.core.date.DateUtil;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;
import java.util.Date;

/**
 * ===自动填充功能===
 * 填充原理是直接给entity的属性设置值!!!
 * MetaObjectHandler提供的默认方法的策略均为:如果属性有值则不覆盖,如果填充值为null则不填充
 * 填充处理器MyMetaObjectHandler在 Spring Boot 中需要声明@Component或@Bean注入
 * 要想根据注解FieldFill.xxx和字段名以及字段类型来区分必须使用父类的strictInsertFill或者strictUpdateFill方法
 * 不需要根据任何来区分可以使用父类的fillStrategy方法
 * update(T t,Wrapper updateWrapper)时t不能为空,否则自动填充失效
 *
 * @author wangMaoXiong
 * @version 1.0
 * @date 2023/4/1 13:18
 */
@Component
public class MyMetaObjectHandler implements MetaObjectHandler {

    /**
     * 新增插入时对公共字段的自动填充。
     * strictInsertFill(MetaObject metaObject, String fieldName, Class<T> fieldType, E fieldVal)
     * * metaObject：待填充的对象。
     * * fieldName：待填充的对象属性名称，如 createTime, updateTime, ...。
     * * fieldType：属性的数据类型，需要兼容不同的数据类型：比如 LocalDateTime、Date、Long 等。
     * * * 属性名称正确，类型不匹配，也是无法正常填充的，比如对象的createTime属性是Long类型，这里却没有提供 Long 类型，则无法自动填充。
     * * fieldVal：填充的值。
     * 也可以使用 setFieldValByName 通用填充 AIP。
     *
     * @param metaObject ：其中包含了对象对象及其值。
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        // 起始版本 3.3.0(推荐使用)
        String[] timeFields = {"createTime", "updateTime"};
        for (String timeField : timeFields) {
            this.strictInsertFill(metaObject, timeField, LocalDateTime.class, LocalDateTime.now());
            this.strictInsertFill(metaObject, timeField, Date.class, DateUtil.date());
            this.strictInsertFill(metaObject, timeField, Long.class, DateUtil.date().getTime());
        }
    }

    /**
     * 更新时对公共字段的自动填充。
     * strictUpdateFill(MetaObject metaObject, String fieldName, Class<T> fieldType, E fieldVal)
     * * metaObject：待填充的对象。
     * * fieldName：待填充的对象属性名称，如 updateTime。
     * * fieldType：属性的数据类型，需要兼容不同的数据类型：比如 LocalDateTime、Date、Long 等。
     * * * 属性名称正确，类型不匹配，也是无法正常填充的，比如对象的createTime属性是Long类型，这里却没有提供 Long 类型，则无法自动填充。
     * * fieldVal：填充的值。
     * 也可以使用 setFieldValByName 通用填充 AIP。
     *
     * @param metaObject ：其中包含了对象对象及其值。
     */
    @Override
    public void updateFill(MetaObject metaObject) {
        // 起始版本 3.3.0(推荐)
        this.strictUpdateFill(metaObject, "updateTime", LocalDateTime.class, LocalDateTime.now());
        this.strictUpdateFill(metaObject, "updateTime", Date.class, DateUtil.date());
        this.strictUpdateFill(metaObject, "updateTime", Long.class, DateUtil.date().getTime());
    }
}